Python conditional iteration